package thread;

import java.util.concurrent.PriorityBlockingQueue;

//通过这个类来描述一个任务
class MyTask implements Comparable<MyTask>{
    //任务要干啥
    private Runnable command;
    //任务啥时候干
    private long time;
    public MyTask(Runnable command,long after){
        this.command=command;
        //此处记录的时间是一个绝对的时间戳，不是“多长时间之后执行”
        this.time=System.currentTimeMillis()+after;

    }
    //执行任务的方法，直接在内部调用Runnable的run即可
    public void run(){
        command.run();
    }

    public long getTime() {
        return time;
    }

    @Override
    public int compareTo(MyTask o) {
        //希望时间小的在前面，时间大的在后面
        return (int)(this.time-o.time);
    }
}
//自己创建的定时器
class MyTimer {
    //这个是用阻塞等待
    private Object locker = new Object();
    //使用优先级队列来保存若干任务
    private PriorityBlockingQueue<MyTask> queue=new PriorityBlockingQueue<>();

    //command 要执行的任务是啥
    //after多长时间之后来执行这个任务

    public void schedule(Runnable command,long after){
        MyTask myTask = new MyTask(command,after);
        synchronized (locker){
            queue.put(myTask);
            locker.notify();
        }

    }
    public MyTimer(){
        //这里启动一个线程
        Thread t = new Thread(()->{
            while (true){
                //每次尝试循环过程中，就不断尝试从队列中获取到队首元素
                //判定队首元素当前的时间是否就绪，如果就绪就执行，不就绪就不执行
                try {
                    synchronized (locker){
                        //queue如果为空的话take操作会死锁
                        if(queue.isEmpty()){
                            locker.wait();
                        }
                    MyTask myTask = queue.take();
                    long curTime = System.currentTimeMillis();
                    if(myTask.getTime()>curTime){
                        //时间还没到，塞回到队列中
                        queue.put(myTask);
                            locker.wait(myTask.getTime()-curTime);

                    }else {
                        //时间到了,直接执行任务
                        myTask.run();
                    }
                    }


                } catch (InterruptedException e) {
                    throw new RuntimeException(e);
                }
            }

        });
        t.start();
    }
}
public class Demo23 {
    public static void main(String[] args) {
        MyTimer myTimer = new MyTimer();
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("111");
            }
        },2000);
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("222");
            }
        },4000);
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("333");
            }
        },6000);
    }
}
